package com.example.mq.mqserver.core;

import com.example.mq.common.mode.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/*
 * 这个类表示一个存储消息的队列
 * MSG => Message
 */
@Data
@Slf4j
public class MSGQueue {
    // 表示队列的身份标识.
    private String name;

    // 表示队列是否持久化, true 表示持久化保存, false 表示不持久化.
    private boolean durable = false;

    // 表示是否被某人独占
    private boolean exclusive = false;

    // 闲置删除 (RabbitMQ 实现了这个功能, 这里未实现)
    private boolean autoDelete = false;

    // 扩展参数 (RabbitMQ 实现了这个功能, 这里未实现)
    private Map<String, Object> arguments = new HashMap<>();

    // 当前队列都有哪些消费者订阅了.
    private List<ConsumerEnv> consumerEnvList = new ArrayList<>();

    // 记录当前取到了第几个消费者. 方便实现轮询策略.
    private AtomicInteger consumerSeq = new AtomicInteger(0);

    // 添加一个新的订阅者
    public void addConsumerEnv(ConsumerEnv consumerEnv) {
        consumerEnvList.add(consumerEnv);
    }

    // 订阅者的删除暂时先不考虑.
    // 挑选一个订阅者, 用来处理当前的消息. (按照轮询的方式)
    public ConsumerEnv chooseConsumer() {
        if (consumerEnvList.isEmpty()) {
            // 该队列没有人订阅
            return null;
        }
        // 计算当前要取的元素的下标.
        int index = consumerSeq.get() % consumerEnvList.size();
        consumerSeq.getAndIncrement();  // +1
        return consumerEnvList.get(index);
    }

    // 将 Map 转成 JSON 格式的字符串，用于和数据库交互
    public String getArguments() {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            log.error(e.getMessage());
            // 异常时返回一个空的 JSON 字符串
            return "{}";
        }
    }

    // 从数据库读取数据后，构造 MSGQueue 对象时会自动调用
    // 将 JSON 字符串解析成 Map 对象
    public void setArguments(String argumentsJson) {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            this.arguments = objectMapper.readValue(argumentsJson, new TypeReference<HashMap<String, Object>>() {});
        } catch (JsonProcessingException e) {
            log.error(e.getMessage());
        }
    }

    public void setArguments(Map<String, Object> arguments) {
        this.arguments = arguments;
    }

    // 提供 getter 和 setter 方法，用来方便地获取/设置 arguments 的键值对 (内部使用)
    public Object getArguments(String key) {
        return arguments.get(key);
    }

    public void setArguments(String key, Object value) {
        arguments.put(key, value);
    }
}
